package com.leyou.trade.service.impl;

import com.leyou.auth.utils.UserContext;
import com.leyou.common.exceptions.LyException;
import com.leyou.trade.entity.CartItem;
import com.leyou.trade.repository.CartItemRepository;
import com.leyou.trade.service.CartService;
import org.springframework.data.domain.Sort;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.data.mongodb.core.query.Update;
import org.springframework.stereotype.Service;
import org.springframework.util.CollectionUtils;

import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.Optional;

/**
 * @author 虎哥
 */
@Service
public class CartServiceImpl implements CartService {

    private final CartItemRepository repository;

    private final MongoTemplate mongoTemplate;

    public CartServiceImpl(CartItemRepository repository, MongoTemplate mongoTemplate) {
        this.repository = repository;
        this.mongoTemplate = mongoTemplate;
    }

    @Override
    public void addCartItem(CartItem cartItem) {
        cartItem = buildCartItem(cartItem);
        repository.save(cartItem);
    }

    private CartItem buildCartItem(CartItem cartItem) {
        // 1.构建出购物车条目的id
        String id = createCartId(cartItem.getSkuId());
        // 2.根据userId和skuId查询当前购物车商品
        Optional<CartItem> optional = repository.findById(id);
        // 3.判断是否存在
        if(optional.isPresent()){
            // 3.1.获取需要新增的数量
            Integer num = cartItem.getNum();
            // 3.2.获取旧的数据
            cartItem = optional.get();
            // 3.3.累加num
            cartItem.setNum(cartItem.getNum() + num);
        }
        // 4.如果不存在，直接新增
        cartItem.setId(id);
        cartItem.setUserId(UserContext.getUser().getId());
        cartItem.setUpdateTime(new Date());
        return cartItem;
    }

    @Override
    public List<CartItem> queryCartList() {
        // 1.获取用户
        Long userId = UserContext.getUser().getId();
        // 2.根据用户查询购物车列表
        List<CartItem> list = repository.queryByUserId(userId, Sort.by("updateTime").descending());
        // 3.判断
        if (CollectionUtils.isEmpty(list)) {
            throw new LyException(404, "购物车为空！");
        }
        return list;
    }

    @Override
    public void updateCartItemNum(Long skuId, Integer num) {
        // 1.获取id
        String id = createCartId(skuId);
        // 2.根据id更新
        mongoTemplate.update(CartItem.class)
                .matching(Query.query(Criteria.where("_id").is(id)))
                .apply(Update.update("num", num))
                .first();
    }

    @Override
    public void addCartItemList(List<CartItem> cartItems) {
        // 准备一个新的item集合
        List<CartItem> insertList = new ArrayList<>(cartItems.size());
        // 循环旧的集合
        for (CartItem cartItem : cartItems) {
            // 处理旧的cartItem，并返回
            CartItem c1 = buildCartItem(cartItem);
            insertList.add(c1);
        }
        // 批量新增
        repository.saveAll(insertList);
    }

    private String createCartId(Long skuId) {
        String template = "u%ds%d";
        return String.format(template, UserContext.getUser().getId(), skuId);
    }
}
